# -*- coding: utf-8 -*-


from gym.envs.registration import register
import logging

# from .Car2DEnv import Car2DEnv

# 根据需要定义其他参数
register(
    id='Car2DEnv-v0',     # 环境名,版本号v0必须有
    entry_point='envs.Car2DEnv:Car2DEnv',      # 文件夹名.文件名:类名
    reward_threshold=9,
)


register(
    id='TestEnv-v0',     # 环境名,版本号v0必须有
    entry_point='envs.TestEnv:TestEnv',      # 文件夹名.文件名:类名
    reward_threshold=1,
)

register(
    id='GridEnv1-v0',     # 环境名,版本号v0必须有
    entry_point='envs.GridEnv1:GridEnv1',      # 文件夹名.文件名:类名
    reward_threshold=100,
    max_episode_steps=200,
)


register(
    id='ImageCartPoleEnv-v0',     # 环境名,版本号v0必须有
    entry_point='envs.ImageCartPoleEnv:ImageCartPoleEnv',      # 文件夹名.文件名:类名
    reward_threshold=200,
    max_episode_steps=500,
)


log = logging.getLogger("custom_env")


def init():
    # logging.basicConfig()
    log.debug("加载自定义环境成功~")
    return 1

